import os
import shutil
import re

# 源路径和目标路径
source_dir = '/home/champrin/Desktop/record_data/网盘/兵种内录/装甲板数据集/2023-联盟赛前/23dataset/check'  # 替换为你的路径
target_dir = '/home/champrin/Desktop/record_data/网盘/兵种内录/装甲板数据集/2023-联盟赛前/23dataset/liu_标注'

# 确保目标目录存在
os.makedirs(target_dir, exist_ok=True)

# 匹配 BigXXX.X 格式的正则表达式
pattern = re.compile(r'^liu.*?\..*$')

# 遍历源目录及其子目录
for root, dirs, files in os.walk(source_dir):
    for file in files:
        # 检查文件名是否符合格式
        if pattern.match(file):
            source_file = os.path.join(root, file)
            target_file = os.path.join(target_dir, file)
            # 复制文件到目标目录
            shutil.move(source_file, target_file)
            print(f'已提取：{source_file} 到 {target_file}')

print('文件提取完成。')
